package com.inspur.dgov.common.tool.node;

import java.util.ArrayList;
import java.util.List;

/**
 * 树节点管理
 *
 * @Author chiyupei
 * @Date 19/9/20 上午10:21
 * @Version 1.0
 * @Description 树节点管理
 */
public class TreeNodeManager<T extends INode> {

    /**
     * 所有节点
     */
    private List<T> list;

    /**
     * 父节点ID
     */
    private List<Long> parentIds = new ArrayList<>();

    public TreeNodeManager(List<T> items) {
        list = items;
    }

    /**
     * 根据节点ID获取一个节点
     * @param id
     * @return
     */
    public INode getTreeNodeAT(Long id) {
        for (INode node : list) {
            if (node.getId().equals(id)) {
                return node;
            }
        }
        return null;
    }

    /**
     * 增加父节点ID
     *
     * @param parentId 父节点ID
     */
    public void addParentId(Long parentId) {
        parentIds.add(parentId);
    }

    /**
     * 获取树的根节点(一个森林对应多颗树)
     *
     * @return 树的根节点集合
     */
    public List<T> getRoot() {
        List<T> roots = new ArrayList<>();
        for (T node : list) {
            if (null == node.getParentId() || parentIds.contains(node.getId())) {
                roots.add(node);
            }
        }
        return roots;
    }
}
